../../../../models/demos/wormhole/stable_diffusion/tests/test_resnet_block_2d.py